Uncomment code for tests