cleanups
Add test assert