Put in a temp test until I've properly written the unit tests