Forgot to export class