Cleanup tests - using assert method instead of assert keyword