Fixed issues in setUp function.