Remove unneeded class construct
Forgot to add test file changes