Small fix for scala