Removed class keyword from tests