Move to chai for testing