[DONE] Fix check for success