Fixed epic fail in code