correction in main activity too