Fix strict typing issue