Fix tests

Fixes tests and hides warnings.