Slight fix

Attempt to get tests working