Changed params in test function
Updated after merging