Fixed errors in test setup