Feat/error handling tests