Fix int type in test class