AI Works, but isn't being called at the correct time yet