Fixed tests

Fixed variable not found error