Bug: removed hardcoded values from test class