Fix if (so it doesn't just work for testing)
Update message as per mbrandt