Add missing tests

Created by: iOiurson